import torch

'''
初始化lr_scheduler时，不设置，last_epoch，
对于那些需要设置last_epoch的，一定是需要调用model的load函数
load函数再对scheduler的last_epoch进行更改。
'''
def init_scheduler(opt, optimizer):
    lr_policy = opt.scheduler.lower()

    if lr_policy == 'none':
        scheduler = None
    elif lr_policy == 'steplr':
        scheduler = torch.optim.lr_scheduler.MultiStepLR(
            optimizer, 
            milestones=opt.milestones,
            gamma=opt.gamma
        )

    return scheduler